how to create numpy array using function